from tfidf.scratch import tfidf

a = [[3, 0, 1],
        [2, 0, 0],
        [3, 0, 0],
        [4, 0, 0],
        [3, 2, 0],
        [3, 0, 2]]

ans = tfidf(a)
print(ans)


aa = [[1, 0], [0, 1]]
bb = [[3, 4] , [4, 3]]
from mat import cos_sim
ans = cos_sim(aa, bb, 0)
print(ans)